python main.py \
    --model opt  \
    --model_args pretrained=facebook/opt-6.7b,model_cache_dir=weight/opt66b_checkpoints,tokenizer_cache_dir=weight/opt66b_tokenizer \
    --tasks piqa \
    --head_importance_calc \
    --save_importance_path logs/head_importance/opt66b/1shot_piqa.pkl \
    --num_fewshot 1